﻿using System;
using System.Collections.Generic;
using System.Data;
using System.Data.Linq.Mapping;
using System.Reflection;

namespace MyCommon
{
    /// <summary>
    /// 数据表功能扩展
    /// </summary>
    public static class DataTableExtentions
    {
        /// <summary>
        /// 转换DataTable中的数据到泛型集合， ——郭旭东
        /// </summary>
        /// <typeparam name="T">要返回数据的强类型指定，类型在定义时成员可指定装饰属性如[Column(Name = "ID")]，以映射DataTable列名称，不指定时表示DataTable列名称与实体属性名称相同</typeparam>
        /// <param name="dt">含有数据的DataTable对象</param>
        /// <returns></returns>
        public static List<T> ConvertToList<T>(this DataTable dt) where T : class, new()
        {
            var list = new List<T>();
            var properties = typeof(T).GetProperties();
            foreach (DataRow item in dt.Rows)
            {
                T model = new T();
                foreach (DataColumn column in dt.Columns)
                {
                    var columnName = column.ColumnName;
                    object columnValue = item[column];
                    try
                    {
                        model.FillProperty(columnName, columnValue, true);
                    }
                    catch (Exception e)
                    {
                        throw new InvalidOperationException(string.Format("在转换泛型集合时，数据字段“{0}”的值“{1}”不能转换到实体属性", columnName, columnValue), e);
                    }
                }
                list.Add(model);
            }
            return list;
        }

        /// <summary>
        /// 转换IDataReader中的数据到泛型集合， ——郭旭东
        /// </summary>
        /// <typeparam name="T">要返回数据的强类型指定，类型在定义时成员可指定装饰属性如[Column(Name = "ID")]，以映射数据库字段名称，不指定时表示数据库字段名称与实体属性名称相同</typeparam>
        /// <param name="reader">含有数据的IDataReader对象</param>
        /// <returns></returns>
        public static List<T> ConvertToList<T>(this IDataReader reader) where T : class, new()
        {
            var list = new List<T>();
            while (reader.Read())
            {
                T model = new T();
                for (int i = 0; i < reader.FieldCount; i++)
                {
                    var columnName = reader.GetName(i);
                    object columnValue = reader[i];
                    try
                    {
                        model.FillProperty(columnName, columnValue, true);
                    }
                    catch (Exception e)
                    {
                        throw new InvalidOperationException(string.Format("在转换泛型集合时，数据字段“{0}”的值“{1}”不能转换到实体属性", columnName, columnValue), e);
                    }
                }
                list.Add(model);
            }
            return list;
        }
    }
}
